package com.t4.common.jwt;


import io.jsonwebtoken.*;
import org.springframework.util.StringUtils;

import java.util.Date;
import java.util.HashMap;

public class JwtHelper {

    // 设置签名有效时长(一周)
    private static long tokenExpiration = 7 * 24 * 60 * 60 * 1000;

    // 设置签名密钥
    private static String tokenSignKey = "t4dmw/b5ge";

    /**
     * 根据传入的数据生成token字符串
     * @param claims 传入的token数据
     * @return
     */
    public static String createToken(HashMap<String, Object> claims){
        String token = Jwts.builder()
                .setSubject("AUTH-USER")
                .setExpiration(new Date(System.currentTimeMillis() + tokenExpiration)) // 有效时长
                .setClaims(claims) // 主体
                .signWith(SignatureAlgorithm.HS256, tokenSignKey) // 签名
                .compressWith(CompressionCodecs.GZIP) // 对字符串进行压缩
                .compact();
        return token;
    }

    /**
     * 获取token中存入的数据
     * @param token
     * @param claimName 数据名称
     * @param tClass 数据类型
     * @param <T>
     * @return
     */
    public static <T> T getClaim(String token, String claimName, Class<T> tClass){
        try{
            if(StringUtils.isEmpty(token)) return null;

            Jws<Claims> claimsJws = Jwts.parser().setSigningKey(tokenSignKey).parseClaimsJws(token);
            Claims claims = claimsJws.getBody();
            T claim = claims.get(claimName, tClass);
            return claim;
        }catch(Exception e){
            e.printStackTrace();
            return null;
        }
    }

    public static void main(String[] args) {
        HashMap<String, Object> map = new HashMap<>();
        map.put("userId", 1L);
        map.put("username", "admin");
        String token = JwtHelper.createToken(map);
        System.out.println(token);

//        Long userId = JwtHelper.getClaim(token, "userId", Long.class);
//        String username = JwtHelper.getClaim(token, "username", String.class);
//        String user = JwtHelper.getClaim(token, "user", String.class);
//        System.out.println(userId);
//        System.out.println(username);
//        System.out.println(user);
    }
}
